package priority_queue;

import java.util.Comparator;
import java.util.PriorityQueue;

// 因为在源码中 PriorityQueue 创建的是小根堆 而这道题中 我们要用到的是大根堆 所以自己重写一下比较器
// 这里的o2.compareTo(o1)就是比较大小的 如果o2大于o1 那么返回正数 反之 则返回负数
class IntCmp implements Comparator<Integer>{

    @Override
    public int compare(Integer o1, Integer o2) {
        return o2.compareTo(o1);
    }
}
public class SmallestK {
    // 有三种解法
    // 1. 直接进行排序 找到最小的 k 个数
    // 2. 整体建立小根堆 前 k 个数就是要找的最小的 k 个数
    // 3. 用前 k 个元素创建一个大根堆 然后将剩余的 N - k个元素将大根堆堆顶的元素进行比较
    //    如果该元素小于堆顶元素 那么就交换该元素和堆顶元素 然后要调整堆(调整堆PriorityQueue会自动调整)


    //使用第三种方法
    public int[] smallestK(int[] arr,int k){
        int[] ret = new int[k];
        if(k == 0 || arr == null){
            return ret;
        }
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k,new IntCmp());
        for(int i = 0;i < k;i++){
            priorityQueue.offer(arr[i]);
        }

        for (int i = k; i < arr.length; i++) {
            int peekVal = priorityQueue.peek();
            if(arr[i] < peekVal){
                priorityQueue.poll();
                priorityQueue.offer(arr[i]);
            }
        }

        for (int i = 0; i < k; i++) {
            ret[i] = priorityQueue.poll();
        }
        return ret;
    }
    // 使用第二种方法
    public int[] smallestK2(int[] arr, int k) {
        int[] ret = new int[k];
        if(k == 0 || arr == null){
            return ret;
        }
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
        for(int i = 0;i < arr.length;i++){
            priorityQueue.offer(arr[i]);
        }

        for (int i = 0; i < k; i++) {
            ret[i] = priorityQueue.poll();
        }
        return ret;
    }
}
